# https://gitee.com/yueyinqiu5990/tj12413601/blob/master/assignment4/question2/main_train_3.py
import torch
import torch.utils.data

import models
import trainer


def main():
    model = models.MultipoleDispersionModel(3)
    trainer.train(model)
    torch.save(model.state_dict(), "./outputs/trained_model_3.pt")


if __name__ == "__main__":
    main()
